There are n
couples sitting in 2n
seats arranged in a row and want to hold hands.
The people and seats are represented by an integer array row
where row[i]
is the ID of the person sitting in the ith
seat. The couples are numbered in order, the first couple being (0, 1)
, the second couple being (2, 3)
, and so on with the last couple being (2n - 2, 2n - 1)
.
Return the minimum number of swaps so that every couple is sitting side by side. A swap consists of choosing any two people, then they stand up and switch seats.
Input: row = [0,2,1,3] Output: 1 Explanation: We only need to swap the second (row[1]) and third (row[2]) person.
Input: row = [3,2,0,1] Output: 0 Explanation: All couples are already seated side by side.
2n == row.length
2 <= n <= 30
n
is even.0 <= row[i] < 2n
- All the elements of
row
are unique.
classSolution: defminSwapsCouples(self, row: List[int]) ->int: row= [person//2forpersoninrow] parent=list(range(len(row) //2)) groups= {} foriinrange(0, len(row), 2): ifrow[i] ==row[i+1]: continuewhileparent[parent[row[i]]] !=parent[row[i]]: parent[row[i]] =parent[parent[row[i]]] whileparent[parent[row[i+1]]] !=parent[row[i+1]]: parent[row[i+1]] =parent[parent[row[i+1]]] ifparent[row[i]] <parent[row[i+1]]: parent[parent[row[i+1]]] =parent[row[i]] else: parent[parent[row[i]]] =parent[row[i+1]] forpersoninparent: whileparent[parent[person]] !=parent[person]: parent[person] =parent[parent[person]] groups[parent[person]] =groups.get(parent[person], 0) +1returnsum(x-1forxingroups.values())